package com.gary;

import com.sun.deploy.util.ArrayUtil;

import java.util.Arrays;

/**
 * 给定一个由 整数 组成的 非空 数组所表示的非负整数，在该数的基础上加一。
 *
 * 最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。
 *
 * 你可以假设除了整数 0 之外，这个整数不会以零开头。
 *
 *
 *
 * 示例 1：
 *
 * 输入：digits = [1,2,3]
 * 输出：[1,2,4]
 * 解释：输入数组表示数字 123。
 *
 * 示例 2：
 *
 * 输入：digits = [4,3,2,1]
 * 输出：[4,3,2,2]
 * 解释：输入数组表示数字 4321。
 *
 * 示例 3：
 *
 * 输入：digits = [0]
 * 输出：[1]
 *
 * @author peng_rui
 * @date 2022/3/6
 */
public class PlusOne {
    public static void main(String[] args) {

        int[] nums = new int[]{9};
        int[] ints = plusOne(nums);
        for (int i = 0; i < ints.length; i++) {
            System.out.print(ints[i]);
        }

    }

    private static int[] plusOne(int[] digits) {
        int len = digits.length;
        for (int i = digits.length - 1; i >= 0; i--) {
            if (digits[i] != 9) {
                digits[i]++;
                return digits;
            } else {
                digits[i] = 0;
            }
        }
        int[] temp = new int[len + 1];
        temp[0] = 1;
        return temp;
    }

    private static int[] plusOneOK1(int[] digits) {
        //首先解决所有数字都为9的情况
        int count9 = 0;
        for (int i = 0; i < digits.length; i++) {
            if (digits[i] == 9)
                count9++;
        }
        if (count9 == digits.length){
            int[] res =  new int[digits.length + 1];
            res[0] = 1;
            return res;
        }

        int len = digits.length - 1;
        do {
            int num = digits[len];
            num++;
            if (num < 10) {
                digits[len] = num;
            } else {
                digits[len] = 0;
            }
            len--;

        } while (digits[len+1] == 0);
        return digits;
    }


    public static int[] plusOneError(int[] digits) {
        double rest = 0;
        for (int i = 0; i < digits.length; i++) {
            int p = digits.length - i - 1;
            int sign = (int) (digits[i] * Math.pow(10, p));
            rest += sign;
        }
        rest++;
        int intRest = (int) rest;
        int[] ints = new int[digits.length + 1];
        int i = 0;
        do {
            int yu = intRest % 10;
            ints[i++] = yu;
            intRest /= 10;
        } while (intRest >= 1);
        int[] ints1 = Arrays.copyOfRange(ints, 0, i);
        int[] ints2 = new int[i];
        for (int j = 0; j < i; j++) {
            ints2[j] = ints1[i - j - 1];
        }
        return ints2;
    }
}
